input_dim = 2560
num_hiddens = 640
num_head = 1
dropout = 0.3
window = 30
lr = 1e-4
epochs = 1000
batch_size = 64
bearing_condition = 2